from . import gen, siliconflow

msg = 'Hi'
messages = gen.make_prompt(user_prompt=msg)
# gen.generate(baidu_deepseek.model_ids[0], messages)

model_ids = [
	# 'Qwen/Qwen3-8B',
	# 'THUDM/GLM-Z1-9B-0414',
	# 'THUDM/GLM-4-9B-0414',
	# 'deepseek-ai/DeepSeek-R1-Distill-Qwen-7B',
	# 'deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B',
	'Qwen/Qwen2.5-7B-Instruct',
	# 'Qwen/Qwen2.5-Coder-7B-Instruct',
	# 'internlm/internlm2_5-7b-chat',
	# 'Qwen/Qwen2-7B-Instruct',
	# 'Qwen/Qwen2-1.5B-Instruct',
	# 'THUDM/glm-4-9b-chat',
	# 'THUDM/chatglm3-6b',
	]
for api_key in siliconflow.api_keys:
	print('\n'+api_key)
	gen.generate(model_ids[0], messages, max_tokens=None, api_key=api_key)

# gen.generate(model_ids[0], messages)